package com.cyan.learn;

import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;

/**
 * 哈希表算法
 */
public class HashTableTest {

    /**
     * 242丶有效的字母异位词
     */
    public boolean isAnagram(String s, String t) {
        int[] arr = new int[26];
        s.chars().forEach(k -> {
            arr[k - 'a'] += 1;
        });

        t.chars().forEach(k -> {
            arr[k - 'a'] -= 1;
        });

        for (int i : arr) {
            if (i != 0) {
                return false;
            }
        }

        return true;
    }


    /**
     * 349丶两个数组的交集
     */
    public int[] intersection(int[] nums1, int[] nums2) {
        HashSet<Integer> set = new HashSet<>();
        HashSet<Integer> set1 = new HashSet<>();

        for (int num : nums1) {
            set.add(num);
        }

        for (int num : nums2) {
            if (set.contains(num)) {
                set1.add(num);
            }
        }

        int[] arr = new int[set1.size()];
        int i = 0;
        for (Integer num : set1) {
            arr[i++] = num;
        }

        return arr;
    }

    /**
     * 454丶四数相加Ⅱ
     */
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        HashMap<Integer, Integer> map = new HashMap<>();

        for (int i : nums1) {
            for (int j : nums2) {
                map.put(i + j, map.getOrDefault(i + j, 0) + 1);
            }
        }

        int count = 0;
        for (int i : nums3) {
            for (int j : nums4) {
                int sum = i + j;
                count += map.getOrDefault(-sum, 0);
            }
        }

        return count;
    }

    /**
     * 383丶赎金信
     */
    public boolean canConstruct(String ransomNote, String magazine) {
        int[] arr = new int[26];

        ransomNote.chars().forEach(s -> {
            arr[s - 'a'] += 1;
        });

        magazine.chars().forEach(s -> {
            arr[s - 'a'] -= 1;
        });

        for (int i : arr) {
            if (i > 0) {
                return false;
            }
        }

        return true;
    }

    /**
     * 15丶三数之和
     * 双指针：左右指针
     */
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);

        int left, right;

        ArrayList<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < nums.length - 2; i++) {
            int num = nums[i];
            if (num > 0) {
                return res;
            }

            if (i > 0 && num == nums[i - 1]) {
                continue;
            }

            left = i + 1;
            right = nums.length - 1;

            while (left < right) {
                int sum = nums[left] + nums[right] + num;
                if (sum == 0) {
                    res.add(Arrays.asList(num, nums[left], nums[right]));
                    // 对右指针去重，即第三个数
                    while (nums[right] == nums[right - 1] && left < right) {
                        right--;
                    }
                    right--;

                    // 对左指针去重，即第二个数
                    while (nums[left] == nums[left + 1] && left < right) {
                        left++;
                    }
                    left++;
                } else if (sum < 0) {
                    left++;
                } else {
                    right--;
                }
            }
        }

        return res;
    }

    /**
     * 18丶四数之和
     */
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);

        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < nums.length - 3; i++) {
            // 剪枝
            if (nums[i] > 0 && target > 0 && nums[i] > target) break;
            // 第一个数去重
            if (i != 0 && nums[i] == nums[i - 1]) continue;

            for (int k = 0; k < nums.length - 2; k++) {
                // 剪枝
                if (nums[k] > 0 && target > 0 && nums[i] + nums[k] > target) break;
                // 第二个数去重
                if (k != i + 1 && nums[k] == nums[k - 1]) continue;

                // 左右指针
                int left = k + 1;
                int right = nums.length - 1;

                while (left < right) {
                    long a = (long) nums[i] + nums[k] + nums[left] + nums[right];
                    if (a == target) {
                        res.add(Arrays.asList(nums[i], nums[k], nums[left], nums[right]));

                        // 对左指针去重
                        while (nums[left] == nums[left + 1] && left < right - 1) {
                            left++;
                        }
                        left++;

                        // 对右指针去重
                        while (nums[right] == nums[right - 1] && left < right - 1) {
                            right--;
                        }
                        right--;
                    } else if (a < target) {
                        left++;
                    } else {
                        right--;
                    }
                }
            }
        }

        return res;
    }
}
